package main

import (
	"fmt"
	"os"
)

type Emp struct {
	Id   int
	Name string
	Next *Emp
}

func (e *Emp) ShowMe() {
	fmt.Printf("链表%d,找到该雇员%d\n", e.Id%7, e.Id)
}

//定义EmpLink
//我们这里的EmpLink不带表头，即第一个节点就存放雇员
type EmpLink struct {
	Head *Emp
}

//添加员工的方法，保证添加是，编号是从小到大
func (el *EmpLink) Insert(emp *Emp) {
	cur := el.Head
	//这是一个辅助指针，在cur前面
	var pre *Emp = nil
	//如果当前的EmpLink，就是一个空链表
	if cur == nil {
		el.Head = emp
		return
	}
	//如果不是一个空链表，给emp找到对应的位置，并插入
	//让cur和emp比较，让pre始终保持在cur前面
	for {
		if cur != nil {
			if cur.Id > emp.Id {
				//找到位置
				break
			}
			pre = cur
			cur = cur.Next
		} else {
			break
		}
	}
	//退出时，我们看一下是否将emp添加到链表的最后

	pre.Next = emp
	emp.Next = cur

}

//显示当前链表的信息
func (el *EmpLink) ShowLink(no int) {
	if el.Head == nil {
		fmt.Printf("链表%d为空\n", no)
		return
	}
	//遍历当前链表，显示数据
	cur := el.Head
	for {
		if cur != nil {
			fmt.Printf("链表%d,雇员id=%d,name=%s  -> ", no, cur.Id, cur.Name)
			cur = cur.Next
		} else {
			break
		}
	}
	fmt.Println()
}

//根据id查找对应的雇员，含有一个链表数组
func (el *EmpLink) FindById(id int) *Emp {
	cur := el.Head
	for {
		if cur != nil && cur.Id == id {
			return cur
		} else if cur == nil {
			break
		}
		cur = cur.Next
	}
	return nil
}

//定义hashtable,含有一个链表数组
type HashTable struct {
	LinkArr [7]EmpLink
}

//给hashtable编写insert雇员的方法
func (ht *HashTable) Insert(emp *Emp) {
	//使用散列函数，确定将该雇员添加到那个链表
	linkNo := ht.HashFun(emp.Id)
	//使用对应的链表添加
	ht.LinkArr[linkNo].Insert(emp)
}

//编写方法，显示hashtable的所有雇员
func (ht *HashTable) ShowAll() {
	for i := 0; i < len(ht.LinkArr); i++ {
		ht.LinkArr[i].ShowLink(i)
	}
}

//编写一个散列方法
func (ht *HashTable) HashFun(id int) int {
	return id % len(ht.LinkArr)
}

//查找方法
func (ht *HashTable) FindById(id int) *Emp {
	linkNo := ht.HashFun(id)
	return ht.LinkArr[linkNo].FindById(id)
}

func main() {
	key := ""
	id := 0
	name := ""
	var hashTable HashTable
	for {
		fmt.Println("============雇员系统踩点=============")
		fmt.Println("input 表示添加雇员")
		fmt.Println("show 表示显示雇员")
		fmt.Println("find 表示查找雇员")
		fmt.Println("exit 表示退出系统")
		fmt.Scanln(&key)
		switch key {
		case "input":
			fmt.Println("输入雇员的id")
			fmt.Scanln(&id)
			fmt.Println("输入雇员的name")
			fmt.Scanln(&name)
			emp := &Emp{
				Id:   id,
				Name: name,
				Next: nil,
			}
			hashTable.Insert(emp)
		case "show":
			hashTable.ShowAll()
		case "exit":
			os.Exit(0)
		case "find":
			fmt.Println("请输入雇员的id号")
			fmt.Scanln(&id)
			emp := hashTable.FindById(id)
			if emp == nil {
				fmt.Printf("id=%d 不存在\n", id)
			} else {
				emp.ShowMe()
			}
		default:
			fmt.Println("输入有误")

		}
	}
}
